home *** CD-ROM | disk | FTP | other *** search
/ Chip 2007 January, February, March & April / Chip-Cover-CD-2007-02.iso / Pakiet bezpieczenstwa / mini Pentoo LiveCD 2006.1 / mpentoo-2006.1.iso / livecd.squashfs / usr / lib / python2.4 / test / test_class.py < prev    next >
Text File  |  2005-10-18  |  7KB  |  371 lines
  1. "Test the functionality of Python classes implementing operators."
  2.  
  3. from test.test_support import TestFailed
  4.  
  5. testmeths = [
  6.  
  7. # Binary operations
  8.     "add",
  9.     "radd",
  10.     "sub",
  11.     "rsub",
  12.     "mul",
  13.     "rmul",
  14.     "div",
  15.     "rdiv",
  16.     "mod",
  17.     "rmod",
  18.     "divmod",
  19.     "rdivmod",
  20.     "pow",
  21.     "rpow",
  22.     "rshift",
  23.     "rrshift",
  24.     "lshift",
  25.     "rlshift",
  26.     "and",
  27.     "rand",
  28.     "or",
  29.     "ror",
  30.     "xor",
  31.     "rxor",
  32.  
  33. # List/dict operations
  34.     "contains",
  35.     "getitem",
  36.     "getslice",
  37.     "setitem",
  38.     "setslice",
  39.     "delitem",
  40.     "delslice",
  41.  
  42. # Unary operations
  43.     "neg",
  44.     "pos",
  45.     "abs",
  46.  
  47. # generic operations
  48.     "init",
  49.     ]
  50.  
  51. # These need to return something other than None
  52. #    "coerce",
  53. #    "hash",
  54. #    "str",
  55. #    "repr",
  56. #    "int",
  57. #    "long",
  58. #    "float",
  59. #    "oct",
  60. #    "hex",
  61.  
  62. # These are separate because they can influence the test of other methods.
  63. #    "getattr",
  64. #    "setattr",
  65. #    "delattr",
  66.  
  67. class AllTests:
  68.     def __coerce__(self, *args):
  69.         print "__coerce__:", args
  70.         return (self,) + args
  71.  
  72.     def __hash__(self, *args):
  73.         print "__hash__:", args
  74.         return hash(id(self))
  75.  
  76.     def __str__(self, *args):
  77.         print "__str__:", args
  78.         return "AllTests"
  79.  
  80.     def __repr__(self, *args):
  81.         print "__repr__:", args
  82.         return "AllTests"
  83.  
  84.     def __int__(self, *args):
  85.         print "__int__:", args
  86.         return 1
  87.  
  88.     def __float__(self, *args):
  89.         print "__float__:", args
  90.         return 1.0
  91.  
  92.     def __long__(self, *args):
  93.         print "__long__:", args
  94.         return 1L
  95.  
  96.     def __oct__(self, *args):
  97.         print "__oct__:", args
  98.         return '01'
  99.  
  100.     def __hex__(self, *args):
  101.         print "__hex__:", args
  102.         return '0x1'
  103.  
  104.     def __cmp__(self, *args):
  105.         print "__cmp__:", args
  106.         return 0
  107.  
  108.     def __del__(self, *args):
  109.         print "__del__:", args
  110.  
  111. # Synthesize AllTests methods from the names in testmeths.
  112.  
  113. method_template = """\
  114. def __%(method)s__(self, *args):
  115.     print "__%(method)s__:", args
  116. """
  117.  
  118. for method in testmeths:
  119.     exec method_template % locals() in AllTests.__dict__
  120.  
  121. del method, method_template
  122.  
  123. # this also tests __init__ of course.
  124. testme = AllTests()
  125.  
  126. # Binary operations
  127.  
  128. testme + 1
  129. 1 + testme
  130.  
  131. testme - 1
  132. 1 - testme
  133.  
  134. testme * 1
  135. 1 * testme
  136.  
  137. if 1/2 == 0:
  138.     testme / 1
  139.     1 / testme
  140. else:
  141.     # True division is in effect, so "/" doesn't map to __div__ etc; but
  142.     # the canned expected-output file requires that __div__ etc get called.
  143.     testme.__coerce__(1)
  144.     testme.__div__(1)
  145.     testme.__coerce__(1)
  146.     testme.__rdiv__(1)
  147.  
  148. testme % 1
  149. 1 % testme
  150.  
  151. divmod(testme,1)
  152. divmod(1, testme)
  153.  
  154. testme ** 1
  155. 1 ** testme
  156.  
  157. testme >> 1
  158. 1 >> testme
  159.  
  160. testme << 1
  161. 1 << testme
  162.  
  163. testme & 1
  164. 1 & testme
  165.  
  166. testme | 1
  167. 1 | testme
  168.  
  169. testme ^ 1
  170. 1 ^ testme
  171.  
  172.  
  173. # List/dict operations
  174.  
  175. 1 in testme
  176.  
  177. testme[1]
  178. testme[1] = 1
  179. del testme[1]
  180.  
  181. testme[:42]
  182. testme[:42] = "The Answer"
  183. del testme[:42]
  184.  
  185. testme[2:1024:10]
  186. testme[2:1024:10] = "A lot"
  187. del testme[2:1024:10]
  188.  
  189. testme[:42, ..., :24:, 24, 100]
  190. testme[:42, ..., :24:, 24, 100] = "Strange"
  191. del testme[:42, ..., :24:, 24, 100]
  192.  
  193.  
  194. # Now remove the slice hooks to see if converting normal slices to slice
  195. # object works.
  196.  
  197. del AllTests.__getslice__
  198. del AllTests.__setslice__
  199. del AllTests.__delslice__
  200.  
  201. import sys
  202. if sys.platform[:4] != 'java':
  203.     testme[:42]
  204.     testme[:42] = "The Answer"
  205.     del testme[:42]
  206. else:
  207.     # This works under Jython, but the actual slice values are
  208.     # different.
  209.     print "__getitem__: (slice(0, 42, None),)"
  210.     print "__setitem__: (slice(0, 42, None), 'The Answer')"
  211.     print "__delitem__: (slice(0, 42, None),)"
  212.  
  213. # Unary operations
  214.  
  215. -testme
  216. +testme
  217. abs(testme)
  218. int(testme)
  219. long(testme)
  220. float(testme)
  221. oct(testme)
  222. hex(testme)
  223.  
  224. # And the rest...
  225.  
  226. hash(testme)
  227. repr(testme)
  228. str(testme)
  229.  
  230. testme == 1
  231. testme < 1
  232. testme > 1
  233. testme <> 1
  234. testme != 1
  235. 1 == testme
  236. 1 < testme
  237. 1 > testme
  238. 1 <> testme
  239. 1 != testme
  240.  
  241. # This test has to be last (duh.)
  242.  
  243. del testme
  244. if sys.platform[:4] == 'java':
  245.     import java
  246.     java.lang.System.gc()
  247.  
  248. # Interfering tests
  249.  
  250. class ExtraTests:
  251.     def __getattr__(self, *args):
  252.         print "__getattr__:", args
  253.         return "SomeVal"
  254.  
  255.     def __setattr__(self, *args):
  256.         print "__setattr__:", args
  257.  
  258.     def __delattr__(self, *args):
  259.         print "__delattr__:", args
  260.  
  261. testme = ExtraTests()
  262. testme.spam
  263. testme.eggs = "spam, spam, spam and ham"
  264. del testme.cardinal
  265.  
  266.  
  267. # return values of some method are type-checked
  268. class BadTypeClass:
  269.     def __int__(self):
  270.         return None
  271.     __float__ = __int__
  272.     __long__ = __int__
  273.     __str__ = __int__
  274.     __repr__ = __int__
  275.     __oct__ = __int__
  276.     __hex__ = __int__
  277.  
  278. def check_exc(stmt, exception):
  279.     """Raise TestFailed if executing 'stmt' does not raise 'exception'
  280.     """
  281.     try:
  282.         exec stmt
  283.     except exception:
  284.         pass
  285.     else:
  286.         raise TestFailed, "%s should raise %s" % (stmt, exception)
  287.  
  288. check_exc("int(BadTypeClass())", TypeError)
  289. check_exc("float(BadTypeClass())", TypeError)
  290. check_exc("long(BadTypeClass())", TypeError)
  291. check_exc("str(BadTypeClass())", TypeError)
  292. check_exc("repr(BadTypeClass())", TypeError)
  293. check_exc("oct(BadTypeClass())", TypeError)
  294. check_exc("hex(BadTypeClass())", TypeError)
  295.  
  296. # mixing up ints and longs is okay
  297. class IntLongMixClass:
  298.     def __int__(self):
  299.         return 0L
  300.  
  301.     def __long__(self):
  302.         return 0
  303.  
  304. try:
  305.     int(IntLongMixClass())
  306. except TypeError:
  307.     raise TestFailed, "TypeError should not be raised"
  308.  
  309. try:
  310.     long(IntLongMixClass())
  311. except TypeError:
  312.     raise TestFailed, "TypeError should not be raised"
  313.  
  314.  
  315. # Test correct errors from hash() on objects with comparisons but no __hash__
  316.  
  317. class C0:
  318.     pass
  319.  
  320. hash(C0()) # This should work; the next two should raise TypeError
  321.  
  322. class C1:
  323.     def __cmp__(self, other): return 0
  324.  
  325. check_exc("hash(C1())", TypeError)
  326.  
  327. class C2:
  328.     def __eq__(self, other): return 1
  329.  
  330. check_exc("hash(C2())", TypeError)
  331.  
  332. # Test for SF bug 532646
  333.  
  334. class A:
  335.     pass
  336. A.__call__ = A()
  337. a = A()
  338. try:
  339.     a() # This should not segfault
  340. except RuntimeError:
  341.     pass
  342. else:
  343.     raise TestFailed, "how could this not have overflowed the stack?"
  344.  
  345.  
  346. # Tests for exceptions raised in instance_getattr2().
  347.  
  348. def booh(self):
  349.     raise AttributeError, "booh"
  350.  
  351. class A:
  352.     a = property(booh)
  353. try:
  354.     A().a # Raised AttributeError: A instance has no attribute 'a'
  355. except AttributeError, x:
  356.     if str(x) != "booh":
  357.         print "attribute error for A().a got masked:", str(x)
  358.  
  359. class E:
  360.     __eq__ = property(booh)
  361. E() == E() # In debug mode, caused a C-level assert() to fail
  362.  
  363. class I:
  364.     __init__ = property(booh)
  365. try:
  366.     I() # In debug mode, printed XXX undetected error and raises AttributeError
  367. except AttributeError, x:
  368.     pass
  369. else:
  370.     print "attribute error for I.__init__ got masked"
  371.